#pragma once
#include "..\MnistDigit\Visitor.h"
#include "Digit.h"

// The following ifdef block is the standard way of creating macros which make exporting 
// from a DLL simpler. All files within this DLL are compiled with the MNISTDIGIT_EXPORTS
// symbol defined on the command line. This symbol should not be defined on any project
// that uses this DLL. This way any other project whose source files include this file see 
// MNISTDIGIT_API functions as being imported from a DLL, whereas this DLL sees symbols
// defined with this macro as being exported.
#ifdef MNISTDIGIT_EXPORTS
#define MNISTDIGIT_API __declspec(dllexport)
#else
#define MNISTDIGIT_API __declspec(dllimport)
#endif

namespace MnistDigit
{
	class MNISTDIGIT_API DigitVisitor : public MnistDigit::Visitor
	{
	public:
		explicit DigitVisitor(std::vector<std::unique_ptr<Digit>>& digits);
		~DigitVisitor(void);

		virtual void Visit(const unsigned char digitValue
			, const std::vector<unsigned char>& data);
	private:
		DigitVisitor& operator=(const DigitVisitor&);
		std::vector<std::unique_ptr<Digit>>& m_digits;
	};

}